package 剑指offer;

import java.util.HashMap;

/**
 * @author: tyy 剑指 Offer
 * 1218. 最长定差子序列
 * 给你一个整数数组 arr 和一个整数 difference，请你找出并返回 arr 中最长等差子序列的长度，该子序列中相邻元素之间的差等于 difference 。
    子序列 是指在不改变其余元素顺序的情况下，通过删除一些元素或不删除任何元素而从 arr 派生出来的序列。
    示例 1：
    输入：arr = [1,2,3,4], difference = 1
    输出：4
    解释：最长的等差子序列是 [1,2,3,4]。
 * @create: 2021-10-31 17:40
 * @description:
 **/
public class Solution16 {
    public int longestSubsequence(int[] arr, int difference) {
        int max = 1;
        HashMap<Integer,Integer> map = new HashMap<Integer,Integer>();
        for(int i=0;i<arr.length;i++){
            Integer val = map.get(arr[i] - difference); // 获取等差子序列前一个元素的值
            if(val != null){
                map.put(arr[i],val+1); // 塞入当前值
                max = max >= val+1 ? max : val+1; // 获取目前能得到最长的等差子序列长度
            }
            else
                map.put(arr[i],1);
        }
        return max;
    }
    public static void main(String[] args) {
        int num = 2;
//        int[] arr = {1,3,5,7,8};
        int[] arr = {1,5,7,8,5,3,4,2,1};

        int result = new Solution16().longestSubsequence(arr, num);
        System.out.println("result = " + result);


    }
}